package cmd

import (
	"ReaderInOutLog/model"
	"os"
	"strconv"
)

var MaxIdStr string

func GetMaxIDFromEnv() int {
	// 从环境变量读取最大 ID
	maxIdStr := os.Getenv("MAX_ID")
	// model.ConnectDB()
	var maxId int

	// 如果环境变量不存在或为空，则从数据库中查询最大 ID
	if maxIdStr == "" {
		maxId, _ = model.GetMaxID()
		os.Setenv("MAX_ID", strconv.Itoa(maxId)) // 将最大 ID 设置为环境变量
		return maxId
	}
	// model.CloseDB()
	maxId, _ = strconv.Atoi(maxIdStr)
	return maxId
}

// 查询数据库中的最大 ID
func SetMaxID(id int) {
	lg.Debugf("SetMaxID: %d", id)
	os.Setenv("MAX_ID", strconv.Itoa(id))
}
